package com.hliushi.mp.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.baomidou.mybatisplus.core.parser.ISqlParser;
import com.baomidou.mybatisplus.extension.parsers.BlockAttackSqlParser;
import com.baomidou.mybatisplus.extension.plugins.OptimisticLockerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import com.baomidou.mybatisplus.extension.plugins.SqlExplainInterceptor;
import com.hliushi.mp.config.id.IdGenerate;
import com.hliushi.mp.config.id.MyMetaObjectHandler;
import com.hliushi.mp.config.id.SnowflakeIdGenerate;
import com.hliushi.mp.injectors.MySqlInjector;
import com.hliushi.mp.plugins.MyInterceptor;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.ArrayList;
import java.util.List;

/**
 * 分页插件配置
 * <p>
 * 注意下: 新版的分页插件已经写法了
 *
 * @author llhuang10
 * @date 2021/8/28 22:01
 */
// todo @MapperScan 一定要这样写吗?, 推断出 包路径一定要指定, mapper接口要有@Repository, 一定吗?
//   Caused by: java.lang.IllegalArgumentException: At least one base package must be specified
@MapperScan(basePackages = "com.hliushi.mp.mapper")
// @MapperScan(basePackages = "com.hliushi.mp.mapper", annotationClass = Repository.class)
@Configuration
public class MybatisPlusConfig {


    /**
     * 配置分页插件
     */
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        return new PaginationInterceptor();
    }


    /**
     * 注入自定义的拦截器 (插件)
     */
    @Bean
    public MyInterceptor myInterceptor() {
        return new MyInterceptor();
    }


    /**
     * sql 分析插件, 可阻断全表更新, 删除的操作
     * 注意点: 该插件仅适用于开发环境, 不使用与生成环境
     * 原因: 有性能损失
     */
    @Bean
    public SqlExplainInterceptor sqlExplainInterceptor() {
        SqlExplainInterceptor sqlExplainInterceptor = new SqlExplainInterceptor();

        List<ISqlParser> list = new ArrayList<>();
        // 全表sql 阻断器
        list.add(new BlockAttackSqlParser());
        sqlExplainInterceptor.setSqlParserList(list);
        return sqlExplainInterceptor;
    }

    /**
     * 设置乐观锁插件
     */
    @Bean
    public OptimisticLockerInterceptor optimisticLockerInterceptor() {
        return new OptimisticLockerInterceptor();
    }

    /**
     * 注入自定义的SQL注入器
     *
     * @return 自定义SQL注入器
     */
    @Bean
    public MySqlInjector mySqlInjector() {
        return new MySqlInjector();
    }

    /**
     * Mybatis Plus 注入器
     * 公共的字段, id, create_user, create_time, update_user, create_time
     *
     * @param idGenerate
     * @return
     */
    @Bean("myMetaObjectHandler")
    public MetaObjectHandler metaObjectHandler(@Qualifier("snowflakeIdGenerate") IdGenerate<Long> idGenerate) {
        return new MyMetaObjectHandler(idGenerate);
    }


    /**
     * id生成 机器码， 单机配置1即可。 集群部署，每个实例自增1即可。
     *
     * @param machineCode
     * @return
     */
    @Bean("snowflakeIdGenerate")
    public IdGenerate getIdGenerate(@Value("${id-generator.machine-code:1}") Long machineCode) {
        return new SnowflakeIdGenerate(machineCode);
    }
}
